package JUC;

import java.util.concurrent.locks.Condition;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;

/**
 * ConditionDemo类用于演示使用Condition来实现线程间的协调同步
 */
public class ConditionDemo {

    /**
     * Resource类表示共享资源，它定义了生产者和消费者线程的操作
     */
    public static class Resource {
        // 资源数量，初始为0
        int num = 0;

        // 定义锁对象
        private Lock lock = new ReentrantLock();
        // 定义两个Condition，分别用于生产者和消费者线程的等待和通知
        private Condition condition1 = lock.newCondition();
        private Condition condition2 = lock.newCondition();

        /**
         * 生产者线程执行的方法，用于添加一个资源
         *
         * @throws InterruptedException 如果线程被中断
         */
        public void add() throws InterruptedException {
            lock.lock();
            try {
                // 当资源数量不为0时，生产者线程等待
                while (num != 0) {
                    condition1.await();
                }
                num++;
                System.out.println(Thread.currentThread().getName() +
                        "生产了1个资源，资源总数为：" + num);
                // 通知消费者线程
                condition2.signal();
            } catch (Exception e) {
                e.printStackTrace();
            } finally {
                lock.unlock();
            }
        }

        /**
         * 消费者线程执行的方法，用于消费一个资源
         *
         * @throws InterruptedException 如果线程被中断
         */
        public void sub() throws InterruptedException {
            lock.lock();
            try {
                // 当资源数量为0时，消费者线程等待
                while (num == 0) {
                    condition2.await();
                }
                num--;
                System.out.println(Thread.currentThread().getName() +
                        "消费了1个资源，资源总数为：" + num);
                // 通知生产者线程
                condition1.signal();
            } catch (Exception e) {
                e.printStackTrace();
            } finally {
                lock.unlock();
            }
        }
    }

    /**
     * 主方法，创建并启动生产者和消费者线程
     *
     * @param args 命令行参数
     */
    public static void main(String[] args) {
        Resource resource = new Resource();
        // 创建并启动生产者线程
        new Thread(() -> {
            for (int i = 0; i < 10; i++) {
                try {
                    resource.add();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        }, "A").start();

        // 创建并启动消费者线程
        new Thread(() -> {
            for (int i = 0; i < 10; i++) {
                try {
                    resource.sub();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        }, "B").start();
    }
}
